package com.mjk.common.core.redis.datatype;

import org.springframework.data.redis.core.Cursor;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ScanOptions;
import org.springframework.data.redis.core.ZSetOperations.TypedTuple;

import java.util.Collection;
import java.util.Set;

/**
 * ZADD key score1 member1 [score2 member2 ...]：向有序集合中添加一个或多个成员，每个成员都关联着一个分数。
 * ZREM key member1 [member2 ...]：从有序集合中移除一个或多个成员。
 * ZRANGE key start stop [WITHSCORES]：按照索引范围获取有序集合中的成员，可以选择是否同时返回成员的分数。
 * ZRANK key member：返回有序集合中指定成员的排名（从小到大）。
 * ZREVRANK key member：返回有序集合中指定成员的排名（从大到小）。
 * ZSCORE key member：返回有序集合中指定成员的分数。
 * ZCOUNT key min max：统计有序集合中分数在指定范围内的成员数量。
 * ZINCRBY key increment member：给有序集合中指定成员的分数增加一个增量值。
 * ZREVRANGE key start stop [WITHSCORES]：按照索引范围获取有序集合中的成员，按照分数从大到小排序，可以选择是否同时返回成员的分数。
 * ZINTERSTORE destination numkeys key1 [key2 ...] [WEIGHTS weight1 [weight2 ...]] [AGGREGATE SUM|MIN|MAX]：计算多个有序集合的交集，并将结果保存到一个新的有序集合中。
 *
 *
 */
public class RedisSortedSetUtils {

	/**
     * 添加元素,有序集合是按照元素的score值由小到大排列
     *
     */
    public static Boolean set(RedisTemplate<String, Object> redisTemplate,String key, String value, double score) {
        return redisTemplate.opsForZSet().add(key, value, score);
    }

    /**
     *
     */
    public static Long zAdd(RedisTemplate<String, Object> redisTemplate,String key, Set<TypedTuple<Object>> values) {
        return redisTemplate.opsForZSet().add(key, values);
    }

    /**
     *
     */
    public static Long zRemove(RedisTemplate<String, Object> redisTemplate,String key, Object... values) {
        return redisTemplate.opsForZSet().remove(key, values);
    }

    /**
     * 增加元素的score值，并返回增加后的值
     *
     */
    public static Double zIncrementScore(RedisTemplate<String, Object> redisTemplate,String key, String value, double delta) {
        return redisTemplate.opsForZSet().incrementScore(key, value, delta);
    }

    /**
     * 返回元素在集合的排名,有序集合是按照元素的score值由小到大排列
     * 
     * @return 0表示第一位
     */
    public static Long zRank(RedisTemplate<String, Object> redisTemplate,String key, Object value) {
        return redisTemplate.opsForZSet().rank(key, value);
    }

    /**
     * 返回元素在集合的排名,按元素的score值由大到小排列
     *
     */
    public static Long zReverseRank(RedisTemplate<String, Object> redisTemplate,String key, Object value) {
        return redisTemplate.opsForZSet().reverseRank(key, value);
    }

    /**
     * 获取集合的元素, 从小到大排序
     * 
     * @param start
     *            开始位置
     * @param end
     *            结束位置, -1查询所有
     */
    public static Set<Object> zRange(RedisTemplate<String, Object> redisTemplate,String key, long start, long end) {
        return redisTemplate.opsForZSet().range(key, start, end);
    }

    /**
     * 获取集合元素, 并且把score值也获取
     *
     */
    public static Set<TypedTuple<Object>> zRangeWithScores(RedisTemplate<String, Object> redisTemplate,String key, long start,
            long end) {
        return redisTemplate.opsForZSet().rangeWithScores(key, start, end);
    }

    /**
     * 根据Score值查询集合元素
     * 
     * @param min
     *            最小值
     * @param max
     *            最大值
     */
    public static Set<Object> zRangeByScore(RedisTemplate<String, Object> redisTemplate,String key, double min, double max) {
        return redisTemplate.opsForZSet().rangeByScore(key, min, max);
    }

    /**
     * 根据Score值查询集合元素, 从小到大排序
     * 
     * @param min
     *            最小值
     * @param max
     *            最大值
     */
    public static Set<TypedTuple<Object>> zRangeByScoreWithScores(RedisTemplate<String, Object> redisTemplate,String key,
            double min, double max) {
        return redisTemplate.opsForZSet().rangeByScoreWithScores(key, min, max);
    }

    /**
     *
     */
    public static Set<TypedTuple<Object>> zRangeByScoreWithScores(RedisTemplate<String, Object> redisTemplate,String key,
            double min, double max, long start, long end) {
        return redisTemplate.opsForZSet().rangeByScoreWithScores(key, min, max,
                start, end);
    }

    /**
     * 获取集合的元素, 从大到小排序
     *
     */
    public static Set<Object> zReverseRange(RedisTemplate<String, Object> redisTemplate,String key, long start, long end) {
        return redisTemplate.opsForZSet().reverseRange(key, start, end);
    }

    /**
     * 获取集合的元素, 从大到小排序, 并返回score值
     *
     */
    public static Set<TypedTuple<Object>> zReverseRangeWithScores(RedisTemplate<String, Object> redisTemplate,String key,
            long start, long end) {
        return redisTemplate.opsForZSet().reverseRangeWithScores(key, start,
                end);
    }

    /**
     * 根据Score值查询集合元素, 从大到小排序
     *
     */
    public static Set<Object> zReverseRangeByScore(RedisTemplate<String, Object> redisTemplate,String key, double min,
            double max) {
        return redisTemplate.opsForZSet().reverseRangeByScore(key, min, max);
    }

    /**
     * 根据Score值查询集合元素, 从大到小排序
     *
     */
    public static Set<TypedTuple<Object>> zReverseRangeByScoreWithScores(RedisTemplate<String, Object> redisTemplate,
            String key, double min, double max) {
        return redisTemplate.opsForZSet().reverseRangeByScoreWithScores(key,
                min, max);
    }

    /**
     *
     */
    public static Set<Object> zReverseRangeByScore(RedisTemplate<String, Object> redisTemplate,String key, double min,
            double max, long start, long end) {
        return redisTemplate.opsForZSet().reverseRangeByScore(key, min, max,
                start, end);
    }

    /**
     * 根据score值获取集合元素数量
     *
     */
    public static Long zCount(RedisTemplate<String, Object> redisTemplate,String key, double min, double max) {
        return redisTemplate.opsForZSet().count(key, min, max);
    }

    /**
     * 获取集合大小
     *
     */
    public static Long zSize(RedisTemplate<String, Object> redisTemplate,String key) {
        return redisTemplate.opsForZSet().size(key);
    }

    /**
     * 获取集合大小
     *
     */
    public static Long zZCard(RedisTemplate<String, Object> redisTemplate,String key) {
        return redisTemplate.opsForZSet().zCard(key);
    }

    /**
     * 获取集合中value元素的score值
     *
     */
    public static Double zScore(RedisTemplate<String, Object> redisTemplate,String key, Object value) {
        return redisTemplate.opsForZSet().score(key, value);
    }

    /**
     * 移除指定索引位置的成员
     *
     */
    public static Long zRemoveRange(RedisTemplate<String, Object> redisTemplate,String key, long start, long end) {
        return redisTemplate.opsForZSet().removeRange(key, start, end);
    }

    /**
     * 根据指定的score值的范围来移除成员
     *
     */
    public static Long zRemoveRangeByScore(RedisTemplate<String, Object> redisTemplate,String key, double min, double max) {
        return redisTemplate.opsForZSet().removeRangeByScore(key, min, max);
    }

    /**
     * 获取key和otherKey的并集并存储在destKey中
     *
     */
    public static Long zUnionAndStore(RedisTemplate<String, Object> redisTemplate,String key, String otherKey, String destKey) {
        return redisTemplate.opsForZSet().unionAndStore(key, otherKey, destKey);
    }

    /**
     *
     */
    public static Long zUnionAndStore(RedisTemplate<String, Object> redisTemplate,String key, Collection<String> otherKeys,
            String destKey) {
        return redisTemplate.opsForZSet()
                .unionAndStore(key, otherKeys, destKey);
    }

    /**
     * 交集
     *
     */
    public static Long zIntersectAndStore(RedisTemplate<String, Object> redisTemplate,String key, String otherKey,
            String destKey) {
        return redisTemplate.opsForZSet().intersectAndStore(key, otherKey,
                destKey);
    }

    /**
     * 交集
     *
     */
    public static Long zIntersectAndStore(RedisTemplate<String, Object> redisTemplate,String key, Collection<String> otherKeys,
            String destKey) {
        return redisTemplate.opsForZSet().intersectAndStore(key, otherKeys,
                destKey);
    }

    /**
     *
     */
    public static Cursor<TypedTuple<Object>> zScan(RedisTemplate<String, Object> redisTemplate,String key, ScanOptions options) {
        return redisTemplate.opsForZSet().scan(key, options);
    }
}
